package com.xbts.common.utils.sign;

import com.xbts.common.exception.ServiceException;
import org.apache.commons.codec.binary.Base64;

import javax.crypto.Cipher;
import java.security.*;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;

/**
 * RSA加密解密
 *
 * @author ruoyi
 **/
public class RsaUtils {

    // Rsa 私钥
    public static String privateKey = "MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDJRSamO8yKJTyY" +
            "veC+pepxM/KUXppg0v4V+IFrPpvI4G42nj/YDow2rMxp2uawLDc8lpHuNAirc4aW" +
            "1li3I8hA6lKBK/n8S6EMNDMgUzwgHd3ruLaFhQefaXCukS+29yF1j50aaSqdbWsT" +
            "wkXViFn5M3Dsu1tN+Qf51THlwdfGgL6wDpGZefCO0+aW7FSecwrbWX+XJse0+kwC" +
            "KkmiTtWnxmB5s6dv5/KYOxn9J4GDSIqAUsH98hMhLHCf0cbWHUnOfYBsu6FZ3Wx+" +
            "dJigE+PYQKJVV4oR0/b0e6E24Pxmj27bk1K8mWDWLnKliY35KMOT/ZuSX4qzNS9n" +
            "OoRtmHUXAgMBAAECggEATC8/tPa3sqbrrAlbk+6rslPug5jXqMTU5q4dFuIpB71k" +
            "GoqGMhs3mc6IxZzVa2QhOG7qKBdMHNHq5ReXlyMVPvUza6+gvY5HFJ6Bp7+555JG" +
            "mV5ewGWdgAriW0GbpQ8gfCZzWjbqqZJsUJtd4bjuDbUaDpHQutAD495+N7DLI1iW" +
            "fS3hddLsM1YrxqYljSlbAx+6C5i/0/Ps+NHKtzoXEBBPjmvxGtnhdD+TiKPUGkZx" +
            "IsVig9CYJBc4svFppzPsfi1Dd+4skXZ01uKf2pI6lhOMS85G9J7FcLxhMei1BaMy" +
            "6W5EGl/8xIKz33qG+opGLyoppNPryGXyJRxiDgptKQKBgQDj1Iq1S3ANXbf5o1GJ" +
            "oQbISo7+e/qRP68CHYXQ+Yf0/7vzSBj49QkQLU0IIUkefpvvA/w4Ry3a/IKDObrl" +
            "kkywH3Upr80WdzTW7K1lUtA5QwAW6iMTod2vz/0hwIqFAoBjSxptsV2jbcPQOuZv" +
            "NfPzgGi7X14/v6CgIVcYaNs8kwKBgQDiJ+hjXyeuGtM47UTpHB/3BUrecbcu7YqZ" +
            "v9Wrg7CivJ/miQG1K1OonavkWM4gKpfphVTXspX81t7RW8oIljjFK3WWI9qBxDxj" +
            "xgWV4jJpBN5G4dgf7bsK2Y33wXH//R4oZrzYHfZud7SXbz72i8wiB+AeUuBtvFzR" +
            "SY6FNFa77QKBgH7Xiy4qRMU8Qzw6ACBWxVbno7rtPXB2ezH7S9AvzNDLPFHS9wxM" +
            "J/KYtgVrurC+noQwvNB8WHxODz+N+Y7sp8N6kBlYZh822PU7qcH6nDD1K2yWiFba" +
            "zHignHaZIsWwwdgGLT2a/N4LM3SQBlEuQZZ1DFodW3S0dHISfdH6Nuz/AoGBALdB" +
            "YUSMiYnCAOBDGzDNcnVm23kOqkIjh8j4MnrNtibYM7XJ1RFp71tVkEDzWOult/0o" +
            "5kv2USVVM65v83Er0QqsIZcsrFW0nr62ajY+2VxR7o9/V/LfX032ZdyZ6vdksrhI" +
            "L8Pr25E9/J8tLNb5BsuYmBnaTDN/i7KnKQ51D8ctAoGBALWtGNBp3X0lsMCjGZ6P" +
            "Xp1Fo7TGEpvpIBL9XxwQoE/x4XD5WLLzQpV/YN/5xEaR1jxjKcC8R3u7oIWN3Xlk" +
            "GVQHlEwtUCO1Cs+Hj11ztWNInXxDabwIUiZ6WaD7a01wfjhsG2BHVrCzBIuEnqft" +
            "YN5aVbqopvHdF8pAble4Awpr";

    /**
     * 私钥解密
     *
     * @param text             待解密的文本
     * @return 解密后的文本
     */
    public static String decryptByPrivateKey(String text){
        try {
            return decryptByPrivateKey(privateKey, text);
        } catch (Exception e) {
            throw new ServiceException("私钥解密失败："+ e.getMessage());
        }
    }

    /**
     * 公钥解密
     *
     * @param publicKeyString 公钥
     * @param text            待解密的信息
     * @return 解密后的文本
     */
    public static String decryptByPublicKey(String publicKeyString, String text) throws Exception {
        X509EncodedKeySpec x509EncodedKeySpec = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 私钥加密
     *
     * @param privateKeyString 私钥
     * @param text             待加密的信息
     * @return 加密后的文本
     */
    public static String encryptByPrivateKey(String privateKeyString, String text) throws Exception {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    /**
     * 私钥解密
     *
     * @param privateKeyString 私钥
     * @param text             待解密的文本
     * @return 解密后的文本
     */
    public static String decryptByPrivateKey(String privateKeyString, String text) throws Exception {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec5 = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec5);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 公钥加密
     *
     * @param publicKeyString 公钥
     * @param text            待加密的文本
     * @return 加密后的文本
     */
    public static String encryptByPublicKey(String publicKeyString, String text) throws Exception {
        X509EncodedKeySpec x509EncodedKeySpec2 = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec2);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    /**
     * 构建RSA密钥对
     *
     * @return 生成后的公私钥信息
     */
    public static RsaKeyPair generateKeyPair() throws NoSuchAlgorithmException {
        KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
        keyPairGenerator.initialize(1024);
        KeyPair keyPair = keyPairGenerator.generateKeyPair();
        RSAPublicKey rsaPublicKey = (RSAPublicKey) keyPair.getPublic();
        RSAPrivateKey rsaPrivateKey = (RSAPrivateKey) keyPair.getPrivate();
        String publicKeyString = Base64.encodeBase64String(rsaPublicKey.getEncoded());
        String privateKeyString = Base64.encodeBase64String(rsaPrivateKey.getEncoded());
        return new RsaKeyPair(publicKeyString, privateKeyString);
    }

    /**
     * RSA密钥对对象
     */
    public static class RsaKeyPair {
        private final String publicKey;
        private final String privateKey;

        public RsaKeyPair(String publicKey, String privateKey) {
            this.publicKey = publicKey;
            this.privateKey = privateKey;
        }

        public String getPublicKey() {
            return publicKey;
        }

        public String getPrivateKey() {
            return privateKey;
        }
    }
}